How to handle edge cases of stack